package com.cz.mydao.dao;

import com.cz.mydao.constant.Constant;
import com.cz.mydao.constant.SqlKey;
import com.cz.mydao.wrapper.SelectWrapper;
import lombok.extern.slf4j.Slf4j;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.List;

/**
 * @descriptions: SQLServer数据库dao层实现类
 * @author: Joker.
 * @date: 2024/12/27 14:45
 * @version: 1.0
 */
@Slf4j
public class SqlServerBaseDao<T> extends AbstractBaseDao<T> {

    /**
     * 分页条件查询
     *
     * @param page     第几页
     * @param pageSize 每页多少条
     * @param wrapper  条件
     * @return 查询结果
     */
    @Override
    public List<T> pageQuery(Long page, Long pageSize, SelectWrapper wrapper){
        if (wrapper == null) {
            return null;
        }
        // 构建分页sql
        StringBuilder sql = buildPageQuerySql(page, pageSize, wrapper);
        log.info(sql.toString());

        try {
            // 获取分页预编译对象
            PreparedStatement preparedStatement = getPreparedStatement(wrapper, sql);
            // 执行查询并返回结果
            return executeQuerySql(preparedStatement);
        } catch (SQLException | InterruptedException e) {
            throw new RuntimeException(Constant.QUERY_FAIL,e);
        }
    }

    /**
     * 获取分页预编译对象
     * @param wrapper 分页条件
     * @param sql 分页sql
     * @return 预编译
     */
    private PreparedStatement getPreparedStatement(SelectWrapper wrapper, StringBuilder sql) throws InterruptedException, SQLException {
        Connection connection = connectionPool.getConnection();
        PreparedStatement preparedStatement = connection.prepareStatement(sql.toString());

        // 设置查询条件的占位符（如果有）
        List<Object> values = wrapper.getValues();
        if (values != null && !values.isEmpty()) {
            for (int i = 0; i < values.size(); i++) {
                // 设置查询条件的占位符
                preparedStatement.setObject(i + 1, values.get(i));
            }
        }
        return preparedStatement;
    }

    /**
     * 构建分页sql
     *
     * @param page     当前页
     * @param pageSize 每页的数量
     * @param wrapper  分页条件
     * @return 分页sql
     */
    private StringBuilder buildPageQuerySql(Long page, Long pageSize, SelectWrapper wrapper) {
        // 构造基本的查询语句（包括查询条件）
        StringBuilder sql = new StringBuilder(wrapper.buildSql(tableName));

        // sqlserver的分页语法必须要有order by条件
        // 如果上面生成sql的时候没有order by条件 ，这俩默认加上按照id排序
        if (wrapper.getOrderBy() == null) {
            sql.append(SqlKey.ORDER_BY).append(SqlKey.ID);
        }

        // 分页条件：计算offset
        long offset = (page - 1) * pageSize;

        // 添加分页语法
        sql.append(SqlKey.OFFSET).append(offset).append(SqlKey.ROWS_FETCH_NEXT).append(pageSize).append(SqlKey.ROWS_ONLY);
        return sql;
    }
}
